package com.e3.web.login.server;

import java.awt.Color;
import java.awt.Font;
import java.awt.Graphics2D;
import java.awt.image.BufferedImage;
import java.io.IOException;
import java.util.Random;

import javax.imageio.ImageIO;
import javax.servlet.ServletException;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import com.e3.web.core.server.HttpSessionHelper;

/**
 * Servlet implementation class for Servlet: ValidateCodeServlet
 * 
 * @web.servlet name="ValidateCodeServlet" display-name="ValidateCodeServlet"
 * @web.servlet-mapping url-pattern="/ValidateCodeServlet"
 * 
 */
public class ValidateCodeServlet extends HttpServlet {
	
	private static final long	serialVersionUID	= 1L;
	
	public final static String	VALIDATE_CODE		= "validate_code";
	
	/** 验证码图片的宽度。 */
	private final static int	width				= 104;
	
	/** 验证码图片的高度。 */
	private final static int	height				= 22;
	
	/** 验证码字符个数 */
	private final static int	codeCount			= 5;
	
	/** 字体宽度 */
	private final static int	fontWidth			= 10;
	
	/** 字体大小 */
	private int					fontSize			= 16;
	
	/** 字体间隔 */
	private int					fontSpace			= 0;
	
	/** 字体离顶端的距离 */
	private int					codeY				= 17;
	
	/** 背景图片地址 */
	private String				bgUrl				= "com/e3/web/login/server/validate_code_bg.jpg";
	
	private final static char[]	codeSequence		= { 'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'J', 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S',
			'T', 'U', 'V', 'W', 'X', 'Y', 'Z', '0', '2', '3', '4', '5', '6', '7', '8', '9', 'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k',
			'm', 'n', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z' };
	
	private final static char[]	simpleCodeSequence	= { '0', '1', '2', '3', '4', '5', '6', '7', '8', '9' };
	
	public void init() throws ServletException {
		fontSpace = (width - codeCount * fontWidth) / (codeCount + 1);
		codeY = height - 5;
	}
	
	@Override
	protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
		doPost(req, resp);
	}
	
	@Override
	protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
		// 创建图像缓冲区
		BufferedImage buffImg = ImageIO.read(ValidateCodeServlet.class.getClassLoader().getResourceAsStream(bgUrl));
		Graphics2D g = buffImg.createGraphics();
		
		// 创建一个随机数生成器类
		Random random = new Random();
		
		// 随机产生干扰线，使图象中的认证码不易被其它程序探测到。
		// g.setColor(Color.BLACK);
		// for (int i = 0; i < 30; i++) {
		// int x = random.nextInt(width);
		// int y = random.nextInt(height);
		// int xl = random.nextInt(12);
		// int yl = random.nextInt(12);
		// g.drawLine(x, y, x + xl, y + yl);
		// }
		
		// randomCode用于保存随机产生的验证码，以便用户登录后进行验证。
		StringBuffer randomCode = new StringBuffer();
		
		// 创建字体，字体的大小
		Font font = new Font("Arial", Font.BOLD, fontSize);
		// 创建字体颜色
		Color color = new Color(44, 103, 149);
		
		// 随机产生codeCount数字的验证码。
		for (int i = 0; i < codeCount; i++) {
			// int red = 0, green = 0, blue = 0;
			// 产生随机的颜色分量来构造颜色值，这样输出的每位数字的颜色值都将不同。
			// red = random.nextInt(255);
			// green = random.nextInt(255);
			// blue = random.nextInt(255);
			// Color color = new Color(red, green, blue);
			// 用随机产生的颜色将验证码绘制到图像中。
			
			// 得到随机产生的验证码数字。
			String strRand = String.valueOf(simpleCodeSequence[random.nextInt(10)]);
			
			g.setFont(font);
			g.setColor(color);
			
			g.drawString(strRand, (i + 1) * fontSpace + i * fontWidth, codeY);
			// 将产生的四个随机数组合在一起。
			randomCode.append(strRand);
		}
		// 将四位数字的验证码保存到Session中。
		HttpSessionHelper.setAttribute(req,VALIDATE_CODE, randomCode.toString());
		// 禁止图像缓存。
		resp.setHeader("Pragma", "no-cache");
		resp.setHeader("Cache-Control", "no-cache");
		resp.setDateHeader("Expires", 0);
		resp.setContentType("image/jpeg");
		// 将图像输出到Servlet输出流中。
		ServletOutputStream sos = resp.getOutputStream();
		ImageIO.write(buffImg, "jpeg", sos);
		sos.close();
	}
	
}
